#!/bin/sh
#
# /etc/init.d/pyspider -- startup script for pyspider
#
#
### BEGIN INIT INFO
# Provides:          pyspider
# Default-Start:     2 3 4 5
# Default-Stop:      0 1 6
### END INIT INFO

PATH=/bin:/usr/bin:/sbin:/usr/sbin
NAME=pyspider
DESC="PySpider Web Crawler"
DEFAULT=/etc/default/$NAME

if [ `id -u` -ne 0 ]; then
	echo "You need root privileges to run this script"
	exit 1
fi


. /lib/lsb/init-functions

if [ -r /etc/default/rcS ]; then
	. /etc/default/rcS
fi


# The following variables can be overwritten in $DEFAULT

# Run pyspider as this user ID and group ID

# Directory where the Elasticsearch binary distribution resides
ES_HOME=/usr/local/bin

# pyspider log directory
LOG_DIR=/var/log/$NAME
LOG_FILE=$LOG_DIR/pyspider.log

# Maximum number of open files
MAX_OPEN_FILES=65535

# pyspider data directory
DATA_DIR=/var/lib/$NAME

# pyspider configuration directory
CONF_DIR=/etc/$NAME
CONF_FILE=$CONF_DIR/config.json

# pyspider PID file directory
PID_DIR="/var/run/pyspider"

# End of variables that can be overwritten in $DEFAULT

# overwrite settings from default file
if [ -f "$DEFAULT" ]; then
	. "$DEFAULT"
fi

# Define other required variables
PID_FILE="$PID_DIR/$NAME.pid"
DAEMON=$ES_HOME/pyspider
MASTER_OPTS="-c $CONF_FILE --log-file $LOG_FILE  --data-path $DATA_DIR all --fetcher-num 2 --processor-num 2  --result-worker-num 2"
SLAVE_OPTS="-c $CONF_FILE  --log-file $LOG_FILE  --data-path $DATA_DIR slave --fetcher-num 2 --processor-num 2  --result-worker-num 2"
MASTER=1
if [ $MASTER -eq 1 ]; then
	DAEMON_OPTS=$MASTER_OPTS
	DESC=$DESC" (Master mode)"
else
	DAEMON_OPTS=$SLAVE_OPTS
	DESC=$DESC" (Slave mode)"
fi

# Check DAEMON exists
test -x $DAEMON || exit 0


case "$1" in
  start)
	log_daemon_msg "Starting $DESC"

	pid=`pidofproc -p $PID_FILE pyspider`
	if [ -n "$pid" ] ; then
		log_begin_msg "Already running."
		log_end_msg 0
		exit 0
	fi

	if [ ! -f "$CONF_FILE" ]; then
		log_failure_msg "Config file $CONF_FILE is not found"
                exit 1
	fi 

	# Prepare environment
	mkdir -p "$LOG_DIR" "$DATA_DIR" "$LOG_DIR"

	# Ensure that the PID_DIR exists (it is cleaned at OS startup time)
	if [ -n "$PID_DIR" ] && [ ! -e "$PID_DIR" ]; then
		mkdir -p "$PID_DIR"
	fi
	if [ -n "$PID_FILE" ] && [ ! -e "$PID_FILE" ]; then
		touch "$PID_FILE"
	fi

	if [ -n "$MAX_OPEN_FILES" ]; then
		ulimit -n $MAX_OPEN_FILES
	fi


	# Start Daemon
	start-stop-daemon --start -b -m --pidfile "$PID_FILE" --exec $DAEMON -- $DAEMON_OPTS
	return=$?
	if [ $return -eq 0 ]; then
		i=0
		timeout=10
		# Wait for the process to be properly started before exiting
		until { cat "$PID_FILE" | xargs kill -0; } >/dev/null 2>&1
		do
			sleep 1
			i=$(($i + 1))
			if [ $i -gt $timeout ]; then
				log_end_msg 1
				exit 1
			fi
		done
	fi
	log_end_msg $return
	exit $return
	;;
  stop)
	log_daemon_msg "Stopping $DESC"

	if [ -f "$PID_FILE" ]; then
		start-stop-daemon --stop --pidfile "$PID_FILE" \
			--quiet \
			--retry forever/TERM/20 > /dev/null
		if [ $? -eq 1 ]; then
			log_progress_msg "$DESC is not running but pid file exists, cleaning up"
		elif [ $? -eq 3 ]; then
			PID="`cat $PID_FILE`"
			log_failure_msg "Failed to stop $DESC (pid $PID)"
			exit 1
		fi
		rm -f "$PID_FILE"
	else
		log_begin_msg "(pyspider is not running)"
	fi
	log_end_msg 0
	;;
  status)
	status_of_proc -p $PID_FILE pyspider pyspider && exit 0 || exit $?
	;;
  restart|force-reload)
	if [ -f "$PID_FILE" ]; then
		$0 stop
		sleep 5
	fi
	$0 start
	;;
  *)
	log_success_msg "Usage: $0 {start|stop|restart|force-reload|status}"
	exit 1
	;;
esac

exit 0
